import torch
import numpy

print(torch.cuda.is_available())
print(torch.version)
print("torchVersion:", torch.__version__)
print("cudaVersion:", torch.version.cuda)
print("cudnnVersion:", torch.backends.cudnn.version())
